package com.dlg.commons.util;

import java.awt.BasicStroke;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.Shape;
import java.awt.geom.RoundRectangle2D;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.OutputStream;
import java.util.Hashtable;

import javax.imageio.ImageIO;

import org.apache.commons.lang3.RandomUtils;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.BinaryBitmap;
import com.google.zxing.DecodeHintType;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatReader;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.Result;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.common.HybridBinarizer;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;

public class QRCodeUtil {

	private static final String CHARSET = "utf-8";
	private static final String FORMAT_NAME = "png";
	private static int QRCODE_SIZE = 300;
	private static final int WIDTH = 60;
	private static final int HEIGHT = 60;
	private static final String LOGO_S = "logo15.png";
	private static final String LOGO_M = "logo30.png";
	private static final String LOGO_L = "logo60.png";
	private static final String LOGO_XL = "logo120.png";

	public static enum QRSize {
		S, // 70x70
		M, // 150x150
		L, // 300x300
		XL // 600x600
	}

	private static BufferedImage createImage(String content, String imgPath,
			boolean needCompress) throws Exception {
		Hashtable<EncodeHintType, Object> hints = new Hashtable<EncodeHintType, Object>();
		hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H);
		hints.put(EncodeHintType.CHARACTER_SET, CHARSET);
		hints.put(EncodeHintType.MARGIN, 1);
		BitMatrix bitMatrix = new MultiFormatWriter().encode(content,
				BarcodeFormat.QR_CODE, QRCODE_SIZE, QRCODE_SIZE, hints);
		/**modify by jgYang 二维码消除白边 start*/
		
		int[] rec = bitMatrix.getEnclosingRectangle();
		int width = rec[2] + 1;  
		int height = rec[3] + 1;  
		BitMatrix resMatrix = new BitMatrix(width, height);  
		resMatrix.clear(); 
		/**modify by jgYang 二维码消除白边 end*/
		
		BufferedImage image = new BufferedImage(width, height,
				BufferedImage.TYPE_INT_RGB);
		for (int x = 0; x < width; x++) {
			for (int y = 0; y < height; y++) {
				image.setRGB(x, y, bitMatrix.get(x + rec[0], y + rec[1]) ? 0xFF000000
						: 0xFFFFFFFF);
			}
		}
		if (imgPath == null || "".equals(imgPath)) {
			return image;
		}
		QRCodeUtil.insertImage(image, imgPath, needCompress);
		return image;
	}

	private static void insertImage(BufferedImage source, String imgPath,
			boolean needCompress) throws Exception {
		File file = new File(imgPath);
		if (!file.exists()) {
			return;
		}
		Image src = ImageIO.read(new File(imgPath));
		int width = src.getWidth(null);
		int height = src.getHeight(null);
		if (needCompress) {
			if (width > WIDTH) {
				width = WIDTH;
			}
			if (height > HEIGHT) {
				height = HEIGHT;
			}
			Image image = src.getScaledInstance(width, height,
					Image.SCALE_SMOOTH);
			BufferedImage tag = new BufferedImage(width, height,
					BufferedImage.TYPE_INT_RGB);
			Graphics g = tag.getGraphics();
			g.drawImage(image, 0, 0, null);
			g.dispose();
			src = image;
		}
		Graphics2D graph = source.createGraphics();
		int x = (QRCODE_SIZE - width) / 2;
		int y = (QRCODE_SIZE - height) / 2;
		graph.drawImage(src, x, y, width, height, null);
		Shape shape = new RoundRectangle2D.Float(x, y, width, width, 6, 6);
		graph.setStroke(new BasicStroke(3f));
		graph.draw(shape);
		graph.dispose();
	}

	public static String encode(String content, String imgPath,
			String destPath, boolean needCompress) throws Exception {
		BufferedImage image = QRCodeUtil.createImage(content, imgPath,
				needCompress);
		mkdirs(destPath);
		String file = + RandomUtils.nextInt(10000000, 99999999) + ".png";
		ImageIO.write(image, FORMAT_NAME, new File(destPath + "/" + file));

		return destPath + "/" + file;
	}

	public static String encode(String content, String destPath,
			boolean haslogo, QRSize size) throws Exception {

		String imgPath = QRCodeUtil.class.getResource("/").getPath();
		String prefix = "";
		int temp = QRCODE_SIZE;
		switch (size) {
		case S:
			QRCODE_SIZE = 70;
			imgPath = haslogo ? imgPath + LOGO_S : null;
			prefix = "S-" + (haslogo ? "T-" : "F-");
			break;
		case M:
			QRCODE_SIZE = 150;
			imgPath = haslogo ? imgPath + LOGO_M : null;
			prefix = "M-" + (haslogo ? "T-" : "F-");
			break;
		case L:
			QRCODE_SIZE = 300;
			imgPath = haslogo ? imgPath + LOGO_L : null;
			prefix = "L-" + (haslogo ? "T-" : "F-");
			break;
		case XL:
			QRCODE_SIZE = 600;
			imgPath = haslogo ? imgPath + LOGO_XL : null;
			prefix = "XL-" + (haslogo ? "T-" : "F-");
			break;
		default:
			QRCODE_SIZE = 150;
			imgPath = haslogo ? imgPath + LOGO_M : null;
			prefix = "M-" + (haslogo ? "T-" : "F-");
			break;
		}

		BufferedImage image = QRCodeUtil.createImage(content, imgPath, true);
		mkdirs(destPath);
		String file = prefix + System.currentTimeMillis() + "-"
				+ RandomUtils.nextInt(100000, 999999) + ".png";
		ImageIO.write(image, FORMAT_NAME, new File(destPath + "/" + file));

		QRCODE_SIZE = temp;
		return destPath + "/" + file;
	}

	public static void mkdirs(String destPath) {
		File file = new File(destPath);
		if (!file.exists() && !file.isDirectory()) {
			file.mkdirs();
		}
	}

	public static void encode(String content, String imgPath, String destPath)
			throws Exception {
		QRCodeUtil.encode(content, imgPath, destPath, false);
	}

	public static void encode(String content, String destPath,
			boolean needCompress) throws Exception {
		QRCodeUtil.encode(content, null, destPath, needCompress);
	}

	public static String encode(String content, String destPath)
			throws Exception {
		return QRCodeUtil.encode(content, null, destPath, false);
	}

	public static void encode(String content, String imgPath,
			OutputStream output, boolean needCompress) throws Exception {
		BufferedImage image = QRCodeUtil.createImage(content, imgPath,
				needCompress);
		ImageIO.write(image, FORMAT_NAME, output);
	}

	public static void encode(String content, OutputStream output)
			throws Exception {
		QRCodeUtil.encode(content, null, output, false);
	}

	public static String decode(File file) throws Exception {
		BufferedImage image;
		image = ImageIO.read(file);
		if (image == null) {
			return null;
		}

		BufferedImageLuminanceSource source = new BufferedImageLuminanceSource(
				image);
		BinaryBitmap bitmap = new BinaryBitmap(new HybridBinarizer(source));
		Result result;
		Hashtable<DecodeHintType, Object> hints = new Hashtable<DecodeHintType, Object>();
		hints.put(DecodeHintType.CHARACTER_SET, CHARSET);
		result = new MultiFormatReader().decode(bitmap, hints);
		String resultStr = result.getText();
		return resultStr;
	}

	public static String decode(String path) throws Exception {
		return QRCodeUtil.decode(new File(path));
	}

	public static String[] encodeAll(String content, String destPath) {
		String[] filelist = new String[6];
		try {
			filelist[0] = QRCodeUtil.encode(content, destPath, false, QRSize.L);
			filelist[1] = QRCodeUtil.encode(content, destPath, false, QRSize.M);
			filelist[2] = QRCodeUtil.encode(content, destPath, false, QRSize.S);

			filelist[3] = QRCodeUtil.encode(content, destPath, true, QRSize.L);
			filelist[4] = QRCodeUtil.encode(content, destPath, true, QRSize.M);
			filelist[5] = QRCodeUtil.encode(content, destPath, true, QRSize.S);
		} catch (Exception e) {
			e.printStackTrace();
		}
		return filelist;
	}

	public static String[] encodeAllNoLogo(String content, String destPath) {
		String[] filelist = new String[3];
		try {
			filelist[0] = QRCodeUtil.encode(content, destPath, false, QRSize.L);
			filelist[1] = QRCodeUtil.encode(content, destPath, false, QRSize.M);
			filelist[2] = QRCodeUtil.encode(content, destPath, false, QRSize.S);
		} catch (Exception e) {
			e.printStackTrace();
		}
		return filelist;
	}
	public static void main(String[] args) {
		String path ="D:\\hx\\newProject\\gongren-project\\gongren-odd-job\\gongren-odd-job-restful-api\\src\\main\\resources\\test\\QRCode";
		try {
			QRCodeUtil.encode("www.baidu.com", path, true, QRSize.M);
		} catch (Exception e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}

}
